# 人口数量 　修改前面的函数，加上第三个必不可
# 少的形参population ，并返回一个格式为 City , Country
# – population xxx 的字符串，如Santiago, Chile –
# population 5000000 。运行test_cities.py，确认测试
# test_city_country() 未通过。
# 修改上述函数，将形参population 设置为可选的。再次运行
# test_cities.py，确认测试test_city_country() 又通过了。
# 再编写一个名为test_city_country_population() 的测
# 试，核实可以使用类似于'santiago' 、'chile'
# 和'population=5000000' 这样的值来调用这个函数。再次
# 运行test_cities.py，确认测试
# test_city_country_population() 通过了。
'''
import unittest 
from city_functions import city_country_population

class CityFunctionsTestCase(unittest.TestCase):
	"""测试city_functions.py"""

	def test_city_country_population(self):
		"""能够正确处理像Santiago Chile这样的城市名和所属国家吗？"""

		city_country_pop = city_country_population('santiago', 'Chile')
		self.assertEqual(city_country_pop, 'Santiago, Chile')


if __name__ == '__main__':
	unittest.main()
'''

# 最终测试
import unittest 
from city_functions import city_country_population_final

class CityFunctionsTestCase(unittest.TestCase):
	"""测试city_functions.py"""

	def test_city_country(self):
		"""能够正确处理像Santiago Chile这样的城市名和所属国家吗？"""

		city_country_pop = city_country_population_final('santiago', 'Chile')
		self.assertEqual(city_country_pop, 'Santiago, Chile')

	def test_city_country_population(self):
		"""
		能够正确处理像Santiago Chile - population 5000000
		这样的城市名和所属国家及其人口吗？
		"""

		city_country_pop = city_country_population_final('santiago', 'Chile', 5000000)
		self.assertEqual(city_country_pop, 'Santiago, Chile - population 5000000')


if __name__ == '__main__':
	unittest.main()
	